package com.sdkj.fixed.asset.common.utils.mail;

/**
 * Mailtest
 *
 * @author shichenxing
 * @Description
 * @date 2020/9/11 12:56
 */

import cn.hutool.core.util.StrUtil;
import com.sun.mail.util.MailSSLSocketFactory;

import javax.activation.DataHandler;
import javax.activation.DataSource;
import javax.activation.FileDataSource;
import javax.mail.*;
import javax.mail.internet.*;
import java.io.UnsupportedEncodingException;
import java.security.GeneralSecurityException;
import java.util.Properties;

/**
 * 邮件发送
 *
 * @author wanglongfei
 * E-mail: islongfei@gmail.com
 * @version 2017年8月27日
 */
public class SendMailWithFile {

    public static void sendMailWithFile(SendMailWithFileConfig entity) throws GeneralSecurityException, MessagingException, UnsupportedEncodingException, AuthenticationFailedException {
        // 收件人电子邮箱
        String to = entity.getTo();

        // 发件人电子邮箱
        String from = entity.getFrom();

        // 指定发送邮件的主机为 smtp.qq.com
        String host = entity.getHost();  //QQ 邮件服务器

        // 获取系统属性
        Properties properties = System.getProperties();

        // 设置邮件服务器
        properties.setProperty("mail.smtp.host", host);

        properties.put("mail.smtp.auth", "true");
        MailSSLSocketFactory sf = new MailSSLSocketFactory();
        sf.setTrustAllHosts(true);
        properties.put("mail.smtp.ssl.enable", "true");
        properties.put("mail.smtp.ssl.socketFactory", sf);
        // 获取默认session对象
        Session session = Session.getInstance(properties, new Authenticator() {
            public PasswordAuthentication getPasswordAuthentication() {     //qq邮箱服务器账户、第三方登录授权码
                return new PasswordAuthentication(entity.getFrom(), entity.getPassword()); //发件人邮件用户名、密码
            }
        });

//        try {
            // 创建默认的 MimeMessage 对象
            MimeMessage message = new MimeMessage(session);

            // Set From: 头部头字段
            message.setFrom(new InternetAddress(from));
            // Set To: 头部头字段
            message.addRecipient(Message.RecipientType.TO, new InternetAddress(to));

            // Set Subject: 主题文字
            message.setSubject(entity.getTitle());

            // 创建消息部分
            BodyPart messageBodyPart = new MimeBodyPart();

            // 消息
            messageBodyPart.setText(entity.getContent());

            // 创建多重消息
            Multipart multipart = new MimeMultipart();

            // 设置文本消息部分
            multipart.addBodyPart(messageBodyPart);

            // 附件部分
            messageBodyPart = new MimeBodyPart();
            //设置要发送附件的文件路径

            if (StrUtil.isNotBlank(entity.getPath())) {

                String filename = entity.getPath();
                DataSource source = new FileDataSource(filename);
                messageBodyPart.setDataHandler(new DataHandler(source));

                //messageBodyPart.setFileName(filename);
                //处理附件名称中文（附带文件路径）乱码问题
                messageBodyPart.setFileName(MimeUtility.encodeText(filename));
                multipart.addBodyPart(messageBodyPart);
            }

            // 发送完整消息
            message.setContent(multipart);

            //   发送消息
            Transport.send(message);

//        } catch (MessagingException | UnsupportedEncodingException mex) {
//            mex.printStackTrace();
//        }
    }

}
